'''
格式会有变化。用openpyxl或xlrd 和 xlwt 库来处理才会保留格式与公式等
'''
import pandas as pd
import os,sys
from pathlib import Path

# 原始 Excel 文件路径
excel_path = os.path.join(Path(__file__).resolve().parent.parent, 'img', 'pmg.xlsx')

# 1. 使用 ExcelFile 读取文件
xl = pd.ExcelFile(excel_path)

# 2. 创建一个字典来存储所有 sheet 的数据
print("可用的工作表名称:")
sheet_dict = {}
for sheet_name in xl.sheet_names:
    sheet_dict[sheet_name] = xl.parse(sheet_name)
    print(f"- {sheet_name}")

# 3. 修改目标 sheet 的数据
target_sheet = '驻点人员'  # 替换为你要修改的 sheet 名称
df = sheet_dict[target_sheet]

# 修改特定单元格（示例：修改第2行第3列的单元格）
#df.iloc[10, 2] = '新值'
df.loc[10, 'K'] = '新值'
# 或者通过列名修改
# df.loc[行索引, '列名'] = '新值'

# 4. 将修改后的数据写回字典
sheet_dict[target_sheet] = df

# 5. 使用 ExcelWriter 保存所有 sheet 回原文件
with pd.ExcelWriter(excel_path, engine='openpyxl') as writer:
    for sheet_name, data in sheet_dict.items():
        data.to_excel(writer, sheet_name=sheet_name, index=False)